// SPDX-License-Identifier: GPL-2.0-only
/*
 * SM3 secure hash, as specified by OSCCA GM/T 0004-2012 SM3 and
 * described at https://tools.ietf.org/html/draft-shen-sm3-hash-01
 *
 * Copyright (C) 2017 ARM Limited or its affiliates.
 * Written by Gilad Ben-Yossef <gilad@benyossef.com>
 * Copyright (C) 2021 Tianjia Zhang <tianjia.zhang@linux.alibaba.com>
 */

#include <crypto/internal/hash.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/mm.h>
#include <linux/types.h>
#include <crypto/sm3.h>
#include <crypto/sm3_base.h>
#include <linux/bitops.h>
#include <asm/byteorder.h>
#include <asm/unaligned.h>

#include "mysm3.h"

const u8 sm3_zero_message_hash[SM3_DIGEST_SIZE] = {
	0x1A, 0xB2, 0x1D, 0x83, 0x55, 0xCF, 0xA1, 0x7F, 0x8e, 0x61, 0x19,
	0x48, 0x31, 0xE8, 0x1A, 0x8F, 0x22, 0xBE, 0xC8, 0xC7, 0x28, 0xFE,
	0xFB, 0x74, 0x7E, 0xD0, 0x35, 0xEB, 0x50, 0x82, 0xAA, 0x2B
};

static void print_buf(u8 *name, u8 *buf, u32 len)
{
	return;

	u32 i;

	pr_err("%s[0x%x]:\n", (char *)name, len);
	for (i = 0; i < len; i += 16) {
		pr_err("%08x: %02x %02x %02x %02x %02x %02x %02x %02x "
		       "%02x %02x %02x %02x %02x %02x %02x %02x\n",
		       i, buf[i + 0], buf[i + 1], buf[i + 2], buf[i + 3],
		       buf[i + 4], buf[i + 5], buf[i + 6], buf[i + 7],
		       buf[i + 8], buf[i + 9], buf[i + 10], buf[i + 11],
		       buf[i + 12], buf[i + 13], buf[i + 14], buf[i + 15]);
	}
}

int crypto_sm3_update(struct shash_desc *desc, const u8 *data, unsigned int len)
{
	print_buf("update data", (u8 *)data, len);
	sm3_update(shash_desc_ctx(desc), data, len);
	return 0;
}

int crypto_sm3_final(struct shash_desc *desc, u8 *out)
{
	sm3_final(shash_desc_ctx(desc), out);
	return 0;
}

int crypto_sm3_finup(struct shash_desc *desc, const u8 *data, unsigned int len,
		     u8 *hash)
{
	struct sm3_state *sctx = shash_desc_ctx(desc);

	if (len)
		sm3_update(sctx, data, len);
	sm3_final(sctx, hash);
	return 0;
}

EXPORT_SYMBOL(crypto_sm3_update);
EXPORT_SYMBOL(crypto_sm3_final);
EXPORT_SYMBOL(crypto_sm3_finup);

void sm3_init_alg_data(struct shash_alg *sm3_alg, struct module *module)
{
	struct shash_alg tmp_sm3_alg = { .digestsize = SM3_DIGEST_SIZE,
					 .init = sm3_base_init,
					 .update = crypto_sm3_update,
					 .final = crypto_sm3_final,
					 .finup = crypto_sm3_finup,
					 .descsize = sizeof(struct sm3_state),
					 .base = {
						 .cra_name = "sm3",
						 .cra_driver_name = "sm33-test",
						 .cra_priority = 500,
						 .cra_blocksize =
							 SM3_BLOCK_SIZE,
						 .cra_module = module,
					 } };

	memcpy(sm3_alg, &tmp_sm3_alg, sizeof(tmp_sm3_alg));
}
EXPORT_SYMBOL(sm3_init_alg_data);

int sm3_generic_mod_init(struct shash_alg *sm3_alg)
{
	return crypto_register_shash(sm3_alg);
}

void sm3_generic_mod_fini(struct shash_alg *sm3_alg)
{
	crypto_unregister_shash(sm3_alg);
}

EXPORT_SYMBOL(sm3_generic_mod_init);
EXPORT_SYMBOL(sm3_generic_mod_fini);

static int _my_init(void)
{
	return 0;
}

static void _my_exit(void)
{
}

subsys_initcall(_my_init);
module_exit(_my_exit);

MODULE_LICENSE("GPL v2");
MODULE_DESCRIPTION("SM3 Secure Hash Algorithm");

MODULE_ALIAS_CRYPTO("sm3");
MODULE_ALIAS_CRYPTO("sm3-generic");
